/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.pegasus.client;

import org.junit.*;

public class TestTimeout {
  @Test
  public void testTimeout() throws PException {
    PegasusClientInterface client = PegasusClientFactory.getSingletonClient();
    String tableName = "temp";

    for (int i = 0; i < 1; ++i) {
      byte[] hashKey = "hello".getBytes();
      byte[] sortKey = "0".getBytes();
      byte[] value = ("world" + String.valueOf(i)).getBytes();

      long current = System.currentTimeMillis();
      try {
        client.set(tableName, hashKey, sortKey, value, 0);
        System.out.println(
            "set succeed, time takes: " + (System.currentTimeMillis() - current) + " ms");
      } catch (PException e) {
        System.out.println(
            "set encounter exception, time takes: "
                + (System.currentTimeMillis() - current)
                + " ms, "
                + e.getMessage());
      }

      current = System.currentTimeMillis();
      try {
        byte[] result = client.get(tableName, hashKey, sortKey);
        System.out.println(
            "get succeed, time takes: " + (System.currentTimeMillis() - current) + " ms");
        if (result != null) {
          System.out.println("get result: " + new String(result));
        }
      } catch (PException e) {
        System.out.println(
            "get encounter exception, time takes: "
                + (System.currentTimeMillis() - current)
                + " ms, "
                + e.getMessage());
        e.printStackTrace();
      }
    }
    PegasusClientFactory.closeSingletonClient();
  }
}
